import torch
import torch.nn as nn
from torchvision import models
import torch.nn.functional as F
model_path = './vgg16-397923af.pth'
model = models.vgg16(pretrained=False)
model.load_state_dict(torch.load(model_path, 'cpu'))
